package com.btrh.generator;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.sql.SQLException;

/**
 * @author zyy
 * @date 2019/10/25 19:30
 * @description
 */
public class MyBatisPlusGenerator {
    public static void main(String[] args) throws SQLException {
        String tableName="a_db_url";//表名，记得创建号数据库表哦
        AutoGenerator mpg = new AutoGenerator();
        // 选择 freemarker 引擎，默认 Veloctiy
        //1. 全局配置
        com.baomidou.mybatisplus.generator.config.GlobalConfig config = new com.baomidou.mybatisplus.generator.config.GlobalConfig();
        config.setActiveRecord(true) // 是否支持AR模式
                .setAuthor("zyy") // 作者
                .setEnableCache(false)// XML 二级缓存
                .setOutputDir("D:\\File\\software\\IDEA_Workspace\\springboot_mybatisplus\\src\\main\\java") // 生成路径
                .setFileOverride(true)  // 文件覆盖
                .setIdType(IdType.AUTO) // 主键策略
                .setBaseResultMap(true)//生成基本的resultMap
                .setBaseColumnList(true)//生成基本的SQL片段
                /* 自定义文件命名，注意 %s 会自动填充表实体属性！ */
                .setMapperName("I%sDao")
                .setXmlName("I%sDao")
                .setServiceName("I%sService")
                .setServiceImplName("%sServiceImpl")
                .setControllerName("%sController");
        mpg.setGlobalConfig(config);

        //2. 数据源配置
        DataSourceConfig dsConfig = new DataSourceConfig();
        dsConfig.setDbType(DbType.MYSQL)  // 设置数据库类型
                .setDriverName("com.mysql.cj.jdbc.Driver")
                .setUrl("jdbc:mysql://192.168.4.212/bt_tunnelbranchcenter?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf-8")
                .setUsername("btsoft")
                .setPassword("btsoft.com");
        mpg.setDataSource(dsConfig);

        //3. 策略配置globalConfiguration中
        StrategyConfig stConfig = new StrategyConfig();
        stConfig.setCapitalMode(true) //全局大写命名
                //.setSuperControllerClass("com.bt.tunnel.controller.BaseController")
                .setNaming(NamingStrategy.underline_to_camel) // 数据库表映射到实体的命名策略
                .setTablePrefix("")//表名前缀
                .setInclude(tableName);  // 需要生成的表

        //4. 包名策略配置
        PackageConfig pkConfig = new PackageConfig();
        pkConfig.setParent("com.btrh")
                .setMapper("mapper")//mapper
                .setService("service")//servcie
                .setController("controller")//controller
                .setEntity("model")
                .setXml("mapper");

        //5. 整合配置
        AutoGenerator ag = new AutoGenerator();
        ag.setGlobalConfig(config)
                .setDataSource(dsConfig)
                .setStrategy(stConfig)
                .setPackageInfo(pkConfig);

        //6. 执行
        ag.execute();
             }

}
